from app.core.database import engine, Base
from app.models.user_config import UserConfig
from app.models.history import History
from loguru import logger

def init_db():
    """初始化数据库"""
    try:
        # 创建所有表
        Base.metadata.create_all(bind=engine)
        logger.info("数据库表创建成功")
        
        # 这里可以添加初始化数据
        # 例如创建默认用户配置等
        
    except Exception as e:
        logger.error(f"数据库初始化失败: {str(e)}")
        raise e

if __name__ == "__main__":
    logger.info("开始初始化数据库...")
    init_db()
    logger.info("数据库初始化完成") 